#!/usr/bin/env python
# -*- conding: utf-8 -*-

"""
@Time     : 2024/8/26 7:02
@Author   : liujingmao
@File     : 1.LLMChain使用技巧.py
"""

import dotenv
from langchain.chains.llm import LLMChain
from langchain_core.prompts import ChatPromptTemplate
from langchain_openai import ChatOpenAI

dotenv.load_dotenv()
prompt = ChatPromptTemplate.from_template("请讲一个关于{subject}主题的冷笑话")
llm = ChatOpenAI(model="gpt-3.5-turbo-16k")

chain = LLMChain(prompt=prompt, llm=llm)

print(chain.invoke({"subject": "中国足球"}))

print("-----------------------------")

print(chain("中国足球"))

print(chain.run("程序员"))
